import torch
import torch.nn as nn

class Model_torch(nn.Module):
    def __init__(self):
        super(Model_torch, self).__init__()
        pass
    def forward(self):
        pass


device = torch.device("cuda:0")

model = Model_torch().to(device)

# 选择随机梯度下降优化器
optimizer = torch.optim.SGD(model.parameters(), lr=4.0)

# 选择优化器步长调节方法StepLR, 用来衰减学习率
"""
第一个参数就是所使用的优化器对象
第二个参数就是每多少轮循环后更新一次学习率(lr)
第三个参数就是每次更新lr的gamma倍
"""
scheduler = torch.optim.lr_scheduler.StepLR(optimizer, 1, gamma=0.9)

